package com.shuaigexue.exam.service.impl;

import com.shuaigexue.exam.config.properties.MinioProperties;
import com.shuaigexue.exam.service.FileUploadService;
import io.minio.*;
import io.minio.errors.*;
import jakarta.annotation.Resource;
import lombok.extern.slf4j.Slf4j;
import org.springframework.stereotype.Service;
import org.springframework.web.multipart.MultipartFile;

import java.io.IOException;
import java.security.InvalidKeyException;
import java.security.NoSuchAlgorithmException;
import java.text.SimpleDateFormat;
import java.util.Date;
import java.util.UUID;

/**
 * @ClassName: FileUploadServiceImpl
 * @Description: 文件上传服务实现类
 }
 }
 }
 * @Author: zhaoxue
 * @Date: 2025/10/28 10:24
 */
@Service
@Slf4j
public class FileUploadServiceImpl implements FileUploadService {


    @Resource
    private MinioClient minioClient;

    @Resource
    private MinioProperties minioProperties;

    @Override
    public String uploadFile(String folder, MultipartFile file) throws ServerException, InsufficientDataException, ErrorResponseException, IOException, NoSuchAlgorithmException, InvalidKeyException, InvalidResponseException, XmlParserException, InternalException {

        // 1.连接minio客户端
        // 2.判断桶是否存在
        boolean bucketedExists = minioClient.bucketExists(BucketExistsArgs.builder().bucket(minioProperties.getBucketName()).build());
        // 3.不存在创建桶，并设置访问权限
        if(!bucketedExists){
            String policy = """
                        {
                              "Statement" : [ {
                                "Action" : "s3:GetObject",
                                "Effect" : "Allow",
                                "Principal" : "*",
                                "Resource" : "arn:aws:s3:::%s/*"
                              } ],
                              "Version" : "2012-10-17"
                        }
                    """.formatted(minioProperties.getBucketName());
            // 创建桶
            minioClient.makeBucket(MakeBucketArgs.builder().bucket(minioProperties.getBucketName()).build());
            // 设置桶访问权限
            minioClient.setBucketPolicy(SetBucketPolicyArgs.builder().bucket(minioProperties.getBucketName()).config(policy).build());
        }
        // 4.上传文件
        // 在minio的服务器中，创建一个文件夹，在存储对象
        // 为了防止重复，uuid进行名字去重
        // 日期banners/ 20251003/uuid_文件名

        String objectName = folder + "/"
                + new SimpleDateFormat("yyyyMMdd").format(new Date()) +"/"
                + UUID.randomUUID().toString().replaceAll("_", "")
                + file.getOriginalFilename();
        minioClient.putObject(PutObjectArgs.builder()
                .bucket(minioProperties.getBucketName()) // 桶名
                .contentType(file.getContentType()) //获取文件类型
                .object(objectName) // 文件名(服务器中对象的名字)
                // stream 上传文件的字节大小
                // 参数1: 就是要上传文件的输入流
                // 参数2: 上传文件大小
                // 参数3: 是否切割文件，转化多线程上传 100 100，写成-1 是由minio自行根据你文件的大小进行切割
                .stream(file.getInputStream(), file.getSize(), -1).build());
        // 5.返回文件访问路径
        //  url = endpoint + "/" + bucketName + "/" + objectName
        String url = minioProperties.getEndpoint() + "/" + minioProperties.getBucketName() + "/" + objectName;
        log.info("文件上传成功，访问路径：{}", url);
        return url;
    }
}
